package com.atguigu.java.ai.langchain4j.tools;

import dev.langchain4j.agent.tool.P;
import dev.langchain4j.agent.tool.Tool;
import dev.langchain4j.agent.tool.ToolMemoryId;
import org.springframework.stereotype.Component;

@Component
public class CalculatorTools {
    //@Tool 注解有两个可选字段：
    //name（工具名称）：工具的名称。如果未提供该字段，方法名会作为工具的名称。
    //value（工具描述）：工具的描述信息。
    //方法参数可以选择使用 @P 注解进行标注。
    //@P
    //注解有两个字段：
    //value：参数的描述信息，这是必填字段。
    //required：表示该参数是否为必需项，默认值为 true ，此为可选字段。
    @Tool(name = "加法", value = "返回两个参数相加之和")
    double sum(
              @ToolMemoryId int memoryId,
              @P(value="加数1", required = true) double a,
              @P(value="加数2", required = true) double b) {
        System.out.println("调用加法运算 " + memoryId);
        return a + b;
    }
    @Tool(name = "平方根", value = "返回给定参数的平方根")
    double squareRoot(double x) {
        System.out.println("调用平方根运算");
        return Math.sqrt(x);
    }
}